LeetCode-28 实现strStr()
给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。
示例 1:
输入: haystack = “hello”, needle = “ll”
输出: 2
示例 2:
输入: haystack = “aaaaa”, needle = “bba”
输出: -1
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
思路
显然,要用两个引用(指针)i,j,从两个字符串头部开始判断。
相同则同时后移(i++,j++),不同则i++。
终止条件:i,j其中之一达到底端。
第一次尝试:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16class Solution {
public int strStr(String haystack, String needle) {
if(needle == null) return 0;
int i = 0, j = 0, n1 = haystack.length(), n2 = needle.length();
while(i < n1 && j < n2){
if(haystack.charAt(i) == needle.charAt(j)){
i++;j++;
continue;
}
i++;
j = 0;
}
if(j == n2) return i-j;
return -1;
}
}
输入: “mississippi” “issip”
输出: -1
预期:4
踩坑错误:忽视了i这个指针,在与j相遇时,应进入一个内循环,终止条件是j到底或i到底,若中途有不相等的,要回到进入位置,故后修改,用一新指针head表示头部。1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23class Solution {
public int strStr(String haystack, String needle) {
if(needle == null) return 0;
int i = 0, j = 0, n1 = haystack.length(), n2 = needle.length();
if(n1 < n2) return -1;
int head = 0;
while(i < n1 && j < n2){
if(haystack.charAt(i) == needle.charAt(j)){
head = i;
while(haystack.charAt(i) == needle.charAt(j)){
if(j == n2-1) return head;
if(i == n1-1) return -1;//这里没写好,处理边界防止i到底移除
i++;j++;
}
i = head;
}
i++;
j = 0;
}
if(j == n2) return head;
return -1;
}
}
看看别人的代码思路。
1 | class Solution { |
狗屁无赖的答案如下:1
2
3
4
5class Solution {
public int strStr(String haystack, String needle) {
return haystack.indexOf(needle);
}
}